"""
配置文件
"""
import os
from dotenv import load_dotenv

# 加载环境变量
load_dotenv()

class Config:
    """应用配置"""
    
    # 数据库配置
    DB_HOST = os.getenv("DB_HOST", "localhost")
    DB_PORT = int(os.getenv("DB_PORT", 3306))
    DB_USER = os.getenv("DB_USER", "root")
    DB_PASSWORD = os.getenv("DB_PASSWORD", "root")
    DB_NAME = os.getenv("DB_NAME", "stock_analysis")
    
    # 数据库连接 URL
    DATABASE_URL = f"mysql+pymysql://{DB_USER}:{DB_PASSWORD}@{DB_HOST}:{DB_PORT}/{DB_NAME}?charset=utf8mb4"
    
    # Redis 配置
    REDIS_HOST = os.getenv("REDIS_HOST", "localhost")
    REDIS_PORT = int(os.getenv("REDIS_PORT", 6379))
    REDIS_DB = int(os.getenv("REDIS_DB", 0))
    
    # API 配置
    API_HOST = os.getenv("API_HOST", "0.0.0.0")
    API_PORT = int(os.getenv("API_PORT", 8000))
    API_DEBUG = os.getenv("API_DEBUG", "True").lower() == "true"
    
    # 缓存配置
    CACHE_EXPIRE_SECONDS = int(os.getenv("CACHE_EXPIRE_SECONDS", 300))
    
    # 日志配置
    LOG_LEVEL = os.getenv("LOG_LEVEL", "INFO")
    
    # 股票 API 配置
    SINA_BASE_URL = "https://hq.sinajs.cn/list="
    NETEASE_HISTORY_URL = "https://quotes.money.163.com/service/chddata.html"
    
    # 热门蓝筹股代码列表（200只）
    POPULAR_STOCKS = [
        # 上证50权重股
        "600519",  # 贵州茅台
        "000858",  # 五粮液
        "601318",  # 中国平安
        "600036",  # 招商银行
        "000651",  # 格力电器
        "002594",  # 比亚迪
        "600809",  # 山西汾酒
        "300750",  # 宁德时代
        "603259",  # 药明康德
        "601888",  # 中国中免
        "600000",  # 浦发银行
        "600887",  # 伊利股份
        "600276",  # 恒瑞医药
        "000002",  # 万科A
        "000001",  # 平安银行
        "601166",  # 兴业银行
        "600030",  # 中信证券
        "000725",  # 京东方A
        "601328",  # 交通银行
        "601012",  # 隆基绿能
        
        # 沪深300核心蓝筹股
        "600104",  # 上汽集团
        "002415",  # 海康威视
        "600585",  # 海螺水泥
        "601398",  # 工商银行
        "601939",  # 建设银行
        "601288",  # 农业银行
        "601988",  # 中国银行
        "600048",  # 保利发展
        "000963",  # 华东医药
        "600031",  # 三一重工
        "000338",  # 潍柴动力
        "002304",  # 洋河股份
        "600690",  # 海尔智家
        "002142",  # 宁波银行
        "600009",  # 上海机场
        "600028",  # 中国石化
        "601857",  # 中国石油
        "601668",  # 中国建筑
        "600050",  # 中国联通
        "000776",  # 广发证券
        
        # 科技龙头股
        "002230",  # 科大讯飞
        "300059",  # 东方财富
        "002475",  # 立讯精密
        "000066",  # 中国长城
        "002027",  # 分众传媒
        "300433",  # 蓝思科技
        "002241",  # 歌尔股份
        "000100",  # TCL科技
        "002352",  # 顺丰控股
        "300014",  # 亿纬锂能
        "002050",  # 三花智控
        "300760",  # 迈瑞医疗
        "688036",  # 传音控股
        "688981",  # 中芯国际
        "002714",  # 牧原股份
        "300015",  # 爱尔眼科
        "000876",  # 新希望
        "002311",  # 海大集团
        "300124",  # 汇川技术
        "002120",  # 韵达股份
        
        # 消费蓝筹股
        "000568",  # 泸州老窖
        "000596",  # 古井贡酒
        "600702",  # 舍得酒业
        "000799",  # 酒鬼酒
        "600132",  # 重庆啤酒
        "600779",  # 水井坊
        "002304",  # 洋河股份
        "600197",  # 伊力特
        "000869",  # 张裕A
        "600559",  # 老白干酒
        "600600",  # 青岛啤酒
        "002493",  # 荣盛石化
        "600297",  # 广汇汽车
        "600258",  # 首旅酒店
        "000895",  # 双汇发展
        "002456",  # 欧菲光
        "000423",  # 东阿阿胶
        "600298",  # 安琪酵母
        "603288",  # 海天味业
        "000729",  # 燕京啤酒
        
        # 医药生物股
        "000538",  # 云南白药
        "600436",  # 片仔癀
        "002821",  # 凯莱英
        "300347",  # 泰格医药
        "002001",  # 新和成
        "300122",  # 智飞生物
        "600161",  # 天坛生物
        "300601",  # 康泰生物
        "000661",  # 长春高新
        "002007",  # 华兰生物
        "603392",  # 万泰生物
        "688185",  # 康希诺
        "600867",  # 通化东宝
        "002422",  # 科伦药业
        "002038",  # 双鹭药业
        "300558",  # 贝达药业
        "300529",  # 健帆生物
        "002030",  # 达安基因
        "300003",  # 乐普医疗
        "002462",  # 嘉事堂
        
        # 新能源汽车产业链
        "002129",  # 中环股份
        "300274",  # 阳光电源
        "002460",  # 赣锋锂业
        "300014",  # 亿纬锂能
        "002466",  # 天齐锂业
        "300438",  # 鹏辉能源
        "002812",  # 恩捷股份
        "300450",  # 先导智能
        "002180",  # 纳思达
        "300073",  # 当升科技
        "300037",  # 新宙邦
        "002709",  # 天赐材料
        "300068",  # 南都电源
        "300316",  # 晶盛机电
        "300751",  # 迈为股份
        "300118",  # 东方日升
        "002074",  # 国轩高科
        "002202",  # 金风科技
        "688005",  # 容百科技
        "002459",  # 晶澳科技
        
        # 金融保险股
        "601601",  # 中国太保
        "601336",  # 新华保险
        "601319",  # 中国人保
        "000783",  # 长江证券
        "002736",  # 国信证券
        "600837",  # 海通证券
        "000166",  # 申万宏源
        "601066",  # 中信建投
        "002797",  # 第一创业
        "600958",  # 东方证券
        "000750",  # 国海证券
        "002926",  # 华西证券
        "600999",  # 招商证券
        "600061",  # 国投资本
        "000686",  # 东北证券
        "600621",  # 华鑫股份
        "601788",  # 光大证券
        "002500",  # 山西证券
        "000728",  # 国元证券
        "600369",  # 西南证券
        
        # 基建建材股
        "601390",  # 中国中铁
        "000425",  # 徐工机械
        "000157",  # 中联重科
        "002152",  # 广电运通
        "600170",  # 上海建工
        "600068",  # 葛洲坝
        "000401",  # 冀东水泥
        "600801",  # 华新水泥
        "002233",  # 塔牌集团
        "000877",  # 天山股份
        "600217",  # 中再资环
        "600761",  # 安徽合力
        "002271",  # 东方雨虹
        "000786",  # 北新建材
        "600309",  # 万华化学
        "000792",  # 盐湖股份
        "600426",  # 华鲁恒升
        "600989",  # 宝丰能源
        "002648",  # 卫星石化
        "000059",  # 华锦股份
        
        # 食品饮料股
        "000568",  # 泸州老窖
        "000596",  # 古井贡酒
        "600132",  # 重庆啤酒
        "600779",  # 水井坊
        "600197",  # 伊力特
        "600559",  # 老白干酒
        "000869",  # 张裕A
        "000729",  # 燕京啤酒
        "600298",  # 安琪酵母
        "000895",  # 双汇发展
        "002216",  # 三全食品
        "000019",  # 深深宝A
        "600300",  # 维维股份
        "002557",  # 洽洽食品
        "000716",  # 黑芝麻
        "603288",  # 海天味业
        "002847",  # 盐津铺子
        "300498",  # 温氏股份
        "000876",  # 新希望
        "002311",  # 海大集团
        
        # 房地产股
        "000002",  # 万科A
        "600048",  # 保利发展
        "001979",  # 招商蛇口
        "000069",  # 华侨城A
        "600383",  # 金地集团
        "600340",  # 华夏幸福
        "000656",  # 金科股份
        "000718",  # 苏宁环球
        "600266",  # 北京城建
        "600895",  # 张江高科
        "000402",  # 金融街
        "600663",  # 陆家嘴
        "000031",  # 中粮地产
        "600240",  # 华业资本
        "000517",  # 荣安地产
        "000540",  # 中天金融
        "000732",  # 泰禾集团
        "000043",  # 中航地产
        "600503",  # 华丽家族
        "000036",  # 华联控股
        
        # 电力公用事业
        "600886",  # 国投电力
        "600795",  # 国电电力
        "000027",  # 深圳能源
        "600021",  # 上海电力
        "600011",  # 华能国际
        "000883",  # 湖北能源
        "600674",  # 川投能源
        "000899",  # 赣能股份
        "600900",  # 长江电力
        "000966",  # 长源电力
        "000875",  # 吉电股份
        "600744",  # 华银电力
        "600236",  # 桂冠电力
        "000531",  # 穗恒运A
        "600726",  # 华电能源
        "600023",  # 浙能电力
        "000600",  # 建投能源
        "000767",  # 漳泽电力
        "600863",  # 内蒙华电
        "000692",  # 惠天热电
    ]

config = Config()

